"""
数据库模型模块
包含所有数据库表的定义
"""

from .user import User
from .data_source import DataSource, DatabaseConfig, ExcelConfig, DataSourceSyncLog, DataSourcePreviewLog
from .analysis_task import AnalysisTask, TaskStatus
from .report import Report, ReportTemplate, ReportStatus
from .system import SystemConfig, OperationLog
from .petition_record import (
    PetitionRecord,
    RegionalAnalysisResult,
    ComplaintTypeAnalysisResult,
    SentimentAnalysisResult,
    ComplianceAnalysisResult,
    DuplicateAnalysisResult
)
from .pollution_type import PollutionType, PollutionTypeExample
from .prompt_template import (
    PromptTemplate,
    PromptParameter,
    PromptUsageRecord,
    PromptCategory,
    PromptTemplateVersion,
    ParameterType,
    OutputFormat
)

__all__ = [
    # 用户相关
    "User",
    # 数据源相关
    "DataSource",
    "DatabaseConfig",
    "ExcelConfig",
    "DataSourceSyncLog",
    "DataSourcePreviewLog",
    
    # 分析任务相关
    "AnalysisTask",
    # "TaskService",
    "TaskStatus",
    
    # 分析结果相关
    # "AnalysisResult",
    # "PetitionData",
    "PollutionType",
    "PollutionTypeExample",
    
    # 信访记录相关
    "PetitionRecord",
    "RegionalAnalysisResult",
    "ComplaintTypeAnalysisResult", 
    "SentimentAnalysisResult",
    "ComplianceAnalysisResult",
    "DuplicateAnalysisResult",
    
    # 报告相关
    "Report",
    "ReportTemplate", 
    "ReportStatus",
    
    # 系统相关
    "SystemConfig",
    "OperationLog",
    
    # 提示词相关
    "PromptTemplate",
    "PromptParameter",
    "PromptUsageRecord",
    "PromptCategory",
    "PromptTemplateVersion",
    "ParameterType",
    "OutputFormat"
]